03. Keras 中的 CNN:实例

Keras 中的 CNN:实际实例

3分27秒处,“对模型训练100个epoch批次大小为32”,意义为“对模型训练100个epoch批次,每个批次大小为32”

如果你尚未通过 AWS 启动 GPU 服务器,强烈建议你先启动,然后再运行此视频中的 notebook。虽然可以在 CPU 上训练 notebook,但是 AWS GPU 实例训练起来将快很多。

你可以在 aind2-cnn GitHub 资源库中访问视频中描述的 Jupyter Notebook。转到 cifar10-classification/ 文件夹并打开 cifar10_mlp.ipynbcifar10_cnn.ipynb

验证集注意事项

在这节课的早些时候,我们通过验证方式训练了一个神经网络,方法是将 model.fit 中的 validation_split 参数设为 0.2。这样会将训练数据的最后 20% 数据变成验证数据。在此视频的 notebook 中,我们自己硬编码了拆分比例,而不是让 Keras 为我们拆分验证集。

课外资料

  • 这是用于在 Keras 中指定神经网络(包括 CNN)的备忘单
  • 参阅 CIFAR-10 竞赛的获胜架构